package JackSu.aobing;

import java.util.HashMap;
import java.util.Map;

/**
 * LRUTest
 *
 * @author Syzer
 * @Description LRU算法
 * @Date 2021/7/20 10:12
 */
public class LRUTest {

    class Node {
        int key;
        int val;
        Node pre;
        Node next;
        public Node(int key, int val) {
            this.key = key;
            this.val = val;
        }
    }

    Node head;
    Node tail;
    int capacity;
    Map<Integer, Node> map;

    public LRUTest(int capacity) {
        this.capacity = capacity;
        map = new HashMap<>();
    }

    public int get(int key) {
        // 获取
        Node node = map.get(key);
        if(node == null) {
            return -1;
        }
        // 放到最新（最后）
        if(node != tail) {
            if (node == head) {
                head = node.next;
            } else {
                node.pre.next = node.next;
                node.next.pre = node.pre;
            }
            node.next = null;
            node.pre = tail;
            tail.next = node;
            tail = node;
        }
        return node.val;
    }

    public void put(int key, int value) {
        if(get(key) == -1){
            // 丢弃最早的
            if (map.size() == capacity) {
                if(map.size() == 1) {
                    map.remove(head.key);
                    head = null;
                    tail = null;
                } else {
                    Node nextHead = head.next;
                    nextHead.pre = null;
                    head.next = null;
                    map.remove(head.key);
                    head = nextHead;
                }
            }
            // 新增
            Node node = new Node(key, value);
            if(tail != null) {
                tail.next = node;
                node.pre = tail;
            } else {
                head = node;
            }
            tail = node;
            map.put(key, node);
        } else {
            // 修改
            tail.key = key;
            tail.val = value;
        }
    }

    public static void main(String[] args) {
        LRUTest lRUCache = new LRUTest(2);
//        lRUCache.put(1, 1); // 缓存是 {1=1}
//        lRUCache.put(2, 2); // 缓存是 {1=1, 2=2}
//        lRUCache.put(3, 3); // 缓存是 {1=1, 2=2}
//        lRUCache.put(4, 4); // 缓存是 {1=1, 2=2}
//        lRUCache.get(4);    // 返回 1
//        lRUCache.get(3);    // 返回 1
//        lRUCache.get(2);    // 返回 1
//        lRUCache.get(1);    // 返回 1
//        lRUCache.put(5, 5); // 该操作会使得关键字 2 作废，缓存是 {1=1, 3=3}
//        lRUCache.get(1);    // 返回 -1 (未找到)
//        lRUCache.get(2);    // 返回 -1 (未找到)
//        lRUCache.get(3);    // 返回 -1 (未找到)
//        lRUCache.get(4);    // 返回 -1 (未找到)
//        lRUCache.get(5);    // 返回 4
        lRUCache.put(1, 1);
        lRUCache.put(2, 2);
        lRUCache.get(1);
        lRUCache.put(3, 3);
        lRUCache.get(2);
        lRUCache.put(4, 4);
        lRUCache.get(1);
        lRUCache.get(3);
        lRUCache.get(4);
    }
}
